import matplotlib.pyplot as plt

from random_walk import RandomWalk

while True:
      
    #创建一个RankWalk实例，并将其包含的点都绘制出来
    rw = RandomWalk(num_points=50000)
    rw.fill_walk()
    #设置窗口尺寸
    plt.figure(figsize=(10, 6))
    point_numbers = list(range(rw.num_points))
    plt.scatter(rw.x_value, rw.y_value, c=point_numbers, cmap=plt.cm.Blues, edgecolors='none', s=1)

    #突出起点和终点
    plt.scatter(0, 0, c='yellow', edgecolors='none',)
    plt.scatter(rw.x_value[-1], rw.y_value[-1], c='red', edgecolors='none', s=100)

    #隐藏坐标轴
    ax = plt.gca()  # 获取当前坐标轴实例
    ax.xaxis.set_visible(False)  # 隐藏 x 轴
    ax.yaxis.set_visible(False)  # 隐藏 y 轴

    plt.show()

    keep_running = input("Make another walk? (y/n)")
    if keep_running == 'n':
        break
    

  
import matplotlib.pyplot as plt
from random_walk import RandomWalk

# 创建一个 RandomWalk 实例
rw = RandomWalk(50000)
rw.fill_walk()

# 使用 plt.plot() 绘制随机漫步轨迹
plt.plot(rw.x_value, rw.y_value, linewidth=1)

# 设置图表的标题，fontsize 参数设置标题字体大小为 24
plt.title("Pollen Movement on Water Droplet Surface", fontsize=24)
# 设置 x 轴的标签，fontsize 参数设置标签字体大小为 14
plt.xlabel('X Position', fontsize=14)
# 设置 y 轴的标签，fontsize 参数设置标签字体大小为 14
plt.ylabel('Y Position', fontsize=14)

# 设置刻度标记的属性，axis='both' 表示同时设置 x 轴和 y 轴，labelsize 设置刻度标签字体大小为 14
plt.tick_params(axis='both', labelsize=14)

# 显示绘制好的图表
plt.show()